package com.hs.dynamic.controller;

import com.hs.dynamic.config.DynamicDataSource;
import com.hs.dynamic.dto.req.UserReq;
import com.hs.dynamic.entity.R;
import com.hs.dynamic.entity.User;
import com.hs.dynamic.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.validation.constraints.NotNull;
import java.util.ArrayList;
import java.util.List;

/**
 * @Description: TODO
 * @Author 胡尚
 * @Date: 2024/8/2 10:12
 */
@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserService userService;

    /**
     * 添加一个dsKey的请求参数，表示使用哪一个数据库
     * @param dsKey 使用哪一个数据库，默认值是hsW
     * @param req 请求参数
     * @return 。
     */
    @PostMapping
    public R addUser(@RequestParam(value = "dsKey",defaultValue = "hsW") String dsKey,
                     @RequestBody UserReq req){
        // 使用哪一个数据库的key 保存至ThreadLocal中
        DynamicDataSource.dataSourceKey.set(dsKey);
        try{
            userService.addUser(req);
        } finally {
            // 请求完成后需要移除
            DynamicDataSource.dataSourceKey.remove();
        }
        return R.ok();
    }

    /**
     * 添加一个dsKey的请求参数，表示使用哪一个数据库
     * @param dsKey 使用哪一个数据库，默认值是hsR
     * @return 。
     */
    @GetMapping
    public R queryUser(@RequestParam(value = "dsKey",defaultValue = "hsR") String dsKey){

        // 使用哪一个数据库的key 保存至ThreadLocal中
        DynamicDataSource.dataSourceKey.set(dsKey);

        List<User> userList = new ArrayList<>();
        try {
            userList.addAll(userService.list());
        } finally {
            // 请求完成后需要移除
            DynamicDataSource.dataSourceKey.remove();
        }
        return R.ok(userList);

    }
}
